import os
import shutil

# 指定包含这些文件的目录
source_directory = 'D:\Model_project\dataprocess\gearbox-picture/test'

# 列出目录中的所有文件
files = os.listdir(source_directory)

for file in files:
    # 忽略非jpg文件
    if not file.endswith('.jpg'):
        continue

    # 提取文件名中的label
    label = file.split('-')[1].split('.')[0]

    # 创建目标文件夹路径
    target_directory = os.path.join(source_directory, label)

    # 如果目标文件夹不存在，则创建它
    if not os.path.exists(target_directory):
        os.makedirs(target_directory)

    # 移动文件
    shutil.move(os.path.join(source_directory, file), target_directory)

print("文件分类完成！")
